$$\mathrm{SageNet}$$: Fast Neural Network Emulation of the Stiff-amplified Gravitational Waves from Inflation
Accurate modeling of the inflationary gravitational waves (GWs) requires time-consuming, iterative numerical integrations of differential equations to take into account their backreaction on the expansion history. To improve computational efficiency while preserving accuracy, we present the Stiff-amplified Gravitational-wave Emulator Network (SageNet), a deep learning framework designed to replace conventional numerical solvers (code available at https://github.com/YifangLuo/SageNet). SageNet employs a long short-term memory architecture to emulate the present-day energy density spectrum of the inflationary GWs with possible stiff amplification, ΩGW(f). Trained on a data set of 25,689 numerically generated solutions, SageNet allows accurate reconstructions of ΩGW(f) and generalizes well to a widemore »